package org.example.graph_theory;

import java.util.Arrays;

public class Redundant_connection {
    public static void main(String[] args) {
        //冗余连接

        //树可以看成是一个连通且 无环 的 无向 图。
        //给定往一棵 n 个节点 (节点值 1～n) 的树中添加一条边后的图。添加的边的两个顶点包含在 1 到 n 中间，且这条附加的边不属于树中已存在的边。图的信息记录于长度为 n 的二维数组 edges ，edges[i] = [ai, bi] 表示图中在 ai 和 bi 之间存在一条边。
        //请找出一条可以删去的边，删除后可使得剩余部分是一个有着 n 个节点的树。如果有多个答案，则返回数组 edges 中最后出现的边。
//        int[][] edges = {{1,2},{1,3},{2,3}};
        int[][] edges = {{1,2},{2,3},{3,4},{1,4},{1,5}};
        int[] redundantConnection = findRedundantConnection(edges);
        System.out.println(Arrays.toString(redundantConnection));
    }

    //并查集算法
    public static int[] findRedundantConnection(int[][] edges) {
        int[] father = new int[edges.length+1];
        int[] deep = new int[edges.length+1];
        int[] redundancy = new int[2];
        for (int i = 1; i <= edges.length; i++) {
            father[i] = i;
        }

        for (int i = 0; i < edges.length; i++) {
            int s = edges[i][0];
            int d = edges[i][1];
            int roots = find(s,father);
            int rootd = find(d, father);
            if (roots == rootd) {
                redundancy = edges[i];
                continue;
            }
            if (deep[roots] > deep[rootd]){
                father[rootd] = roots;
            }else if (deep[roots] < deep[rootd]){
                father[roots] = rootd;
            }else {
                father[rootd] = roots;
                deep[roots]++;
            }
        }
        return redundancy;
    }

    public static int find(int node,int[] father){
        if (father[node] == node) return node;
        return find(father[node],father);
    }
}
